diff --git a/test/setup.jl b/test/setup.jl
index 6a53f725a..a11159623 100644
--- a/test/setup.jl
+++ b/test/setup.jl
@@ -4,5 +4,6 @@ using CairoMakie
 using GeometryBasics
 using DataFrames
 using TopoPlots
+using Colors
 
 include("../docs/example_data.jl")
\ No newline at end of file
diff --git a/test/test_butterfly.jl b/test/test_butterfly.jl
index cfcabc8fc..fcf1985e5 100644
--- a/test/test_butterfly.jl
+++ b/test/test_butterfly.jl
@@ -1,7 +1,6 @@
 
 include("../docs/example_data.jl")
 data, pos = example_data("TopoPlots.jl")
-results, positions = example_data("TopoPlots.jl")
 using Colors
 
 @testset "basic" begin
@@ -18,27 +17,26 @@ end
     )
 end
 
-
 @testset "changing color from ROMA to gray" begin
     plot_butterfly(
-        results;
-        positions = positions,
+        data;
+        positions = pos,
         topopositions_to_color = x -> Colors.RGB(0.5)
     )
 end
 
 @testset "changing color from ROMA to HSV" begin
     plot_butterfly(
-        results;
-        positions = positions,
+        data;
+        positions = pos,
         topopositions_to_color=UnfoldMakie.posToColorHSV
     )
 end
 
 @testset "changing color from ROMA to RGB" begin
     plot_butterfly(
-        results;
-        positions = positions,
+        data;
+        positions = pos,
         topopositions_to_color = pos -> UnfoldMakie.posToColorRGB(pos)
     )
 end