diff --git a/pkg/deppy/solver/solver_test.go b/pkg/deppy/solver/solver_test.go
index 32026dd..dd8694c 100644
--- a/pkg/deppy/solver/solver_test.go
+++ b/pkg/deppy/solver/solver_test.go
@@ -88,8 +88,7 @@ var _ = Describe("Entity", func() {
 		variables := []deppy.Variable{
 			input.NewSimpleVariable("1", constraint.Mandatory()),
 		}
-		s := NewEntitySource(variables)
-		so := solver.NewDeppySolver(s, s)
+		so := solver.NewDeppySolver(&VariableSourceStruct{variables: variables})
 		solution, err := so.Solve(context.Background())
 		Expect(err).ToNot(HaveOccurred())
 		Expect(solution).ToNot(BeNil())