diff --git a/src/GraphQL/ObjectGraphQlEx.php b/src/GraphQL/ObjectGraphQlEx.php
index 5eac3fb..35f19bd 100644
--- a/src/GraphQL/ObjectGraphQlEx.php
+++ b/src/GraphQL/ObjectGraphQlEx.php
@@ -35,7 +35,7 @@ class ObjectGraphQlEx extends Extension
 
         //$var = URLLinkablePlugin::config()->get('single_field_name');
         $var = 'url';
-        if ($curr::class === GraphQLController::class) {
+        if (get_class($curr) === GraphQLController::class) {
             $vars = json_decode($curr->getRequest()->getBody(), true)['variables'];
             if (isset($vars[$var])) {
                 $link = $vars[$var];
diff --git a/src/GraphQL/URLLinkablePlugin.php b/src/GraphQL/URLLinkablePlugin.php
index 1b3d287..0cf3277 100644
--- a/src/GraphQL/URLLinkablePlugin.php
+++ b/src/GraphQL/URLLinkablePlugin.php
@@ -18,6 +18,7 @@ use SilverStripe\GraphQL\Schema\Schema;
 use SilverStripe\ORM\ArrayList;
 use SilverStripe\View\ArrayData;
 use SilverStripe\View\SSViewer;
+use SilverStripe\View\ViewableData;
 
 if (!interface_exists(ModelQueryPlugin::class)) {
     return;
@@ -219,7 +220,7 @@ class URLLinkablePlugin implements ModelQueryPlugin
 
     protected static function findAction($controller, $request)
     {
-        $handlerClass = $controller::class;
+        $handlerClass = get_class($controller);
 
         // We stop after RequestHandler; in other words, at ViewableData
         while ($handlerClass && $handlerClass != ViewableData::class) {