diff --git a/src/FormFields/SessionManagerField.php b/src/FormFields/SessionManagerField.php
index f45eb60..eaaaa9f 100644
--- a/src/FormFields/SessionManagerField.php
+++ b/src/FormFields/SessionManagerField.php
@@ -8,7 +8,7 @@
 use SilverStripe\Security\Member;
 use SilverStripe\SessionManager\Controllers\LoginSessionController;
 use SilverStripe\SessionManager\Models\LoginSession;
-use SilverStripe\View\ViewableData;
+use SilverStripe\Model\ModelData;
 
 class SessionManagerField extends FormField
 {
@@ -26,7 +26,7 @@ class SessionManagerField extends FormField
      * {@inheritDoc}
      *
      * @param string $name Field name
-     * @param null|string|ViewableData $title Field title
+     * @param null|string|ModelData $title Field title
      * @param mixed $value Member ID to apply this field to
      * @param string $titleLinkText Title link text
      * @param string $titleLinkHref Title link href