diff --git a/tests/lib/User/SessionTest.php b/tests/lib/User/SessionTest.php
index f0f016e794af4..156c9b59f23b9 100644
--- a/tests/lib/User/SessionTest.php
+++ b/tests/lib/User/SessionTest.php
@@ -14,6 +14,7 @@
 use OC\Authentication\Exceptions\PasswordLoginForbiddenException;
 use OC\Authentication\Token\IProvider;
 use OC\Authentication\Token\IToken;
+use OC\Authentication\Token\PublicKeyToken;
 use OC\Security\Bruteforce\Throttler;
 use OC\Session\Memory;
 use OC\User\LoginException;