diff --git a/features/steps/project/project_network_graph.rb b/features/steps/project/project_network_graph.rb
index f26deff9..b66aadfe 100644
--- a/features/steps/project/project_network_graph.rb
+++ b/features/steps/project/project_network_graph.rb
@@ -11,7 +11,7 @@ class ProjectNetworkGraph < Spinach::FeatureSteps
 
   And 'I visit project "Shop" network page' do
     # Stub Graph::JsonBuilder max_size to speed up test (10 commits vs. 650)
-    Gitlab::Graph::JsonBuilder.stub(max_count: 10)
+    Graph::JsonBuilder.stub(max_count: 10)
 
     project = Project.find_by_name("Shop")
     visit project_graph_path(project, "master")
diff --git a/features/steps/shared/paths.rb b/features/steps/shared/paths.rb
index 40786f6e..431d5299 100644
--- a/features/steps/shared/paths.rb
+++ b/features/steps/shared/paths.rb
@@ -143,7 +143,7 @@ module SharedPaths
 
   Given "I visit my project's network page" do
     # Stub Graph::JsonBuilder max_size to speed up test (10 commits vs. 650)
-    Gitlab::Graph::JsonBuilder.stub(max_count: 10)
+    Graph::JsonBuilder.stub(max_count: 10)
 
     visit project_graph_path(@project, root_ref)
   end