1 files changed, 3 insertions, 3 deletions
diff --git a/lib/gitlab/import_export/project/sample/relation_tree_restorer.rb b/lib/gitlab/import_export/project/sample/relation_tree_restorer.rb
index 4db92b12968..034122a9f14 100644
--- a/lib/gitlab/import_export/project/sample/relation_tree_restorer.rb
+++ b/lib/gitlab/import_export/project/sample/relation_tree_restorer.rb
@@ -4,7 +4,7 @@ module Gitlab
   module ImportExport
     module Project
       module Sample
-        class RelationTreeRestorer < ImportExport::RelationTreeRestorer
+        class RelationTreeRestorer < ImportExport::Project::RelationTreeRestorer
           def initialize(...)
             super(...)
 
@@ -18,10 +18,10 @@ module Gitlab
           end
 
           def dates
-            return [] if relation_reader.legacy?
+            return [] if @relation_reader.legacy?
 
             RelationFactory::DATE_MODELS.flat_map do |tag|
-              relation_reader.consume_relation(@importable_path, tag, mark_as_consumed: false).map do |model|
+              @relation_reader.consume_relation(@importable_path, tag, mark_as_consumed: false).map do |model|
                 model.first['due_date']
               end
             end