diff --git a/core/generate/context_test.go b/core/generate/context_test.go
index a6cb034..644cb94 100644
--- a/core/generate/context_test.go
+++ b/core/generate/context_test.go
@@ -114,3 +114,26 @@ func TestGenerateContext(t *testing.T) {
 
 	snaps.MatchJSON(t, serializedPlan)
 }
+
+func TestPython(t *testing.T) {
+	ctx := CreateTestContext(t, "../../examples/node-npm")
+	provider := &TestProvider{}
+	require.NoError(t, provider.Plan(ctx))
+
+	// User defined config
+	configJSON := `{
+		"packages": {
+			"python": "3.11"
+		}
+	}`
+
+	var cfg config.Config
+	require.NoError(t, json.Unmarshal([]byte(configJSON), &cfg))
+
+	// Apply the config to the context
+	require.NoError(t, ctx.ApplyConfig(&cfg))
+
+	// Resolve packages
+	_, err := ctx.ResolvePackages()
+	require.NoError(t, err)
+}
diff --git a/core/resolver/resolver_test.go b/core/resolver/resolver_test.go
index 09d1028..2c8d829 100644
--- a/core/resolver/resolver_test.go
+++ b/core/resolver/resolver_test.go
@@ -94,17 +94,3 @@ func PythonTest1(t *testing.T) {
 	require.NoError(t, err)
 	assert.Equal(t, 1, len(resolvedPackages))
 }
-
-func PythonTest2(t *testing.T) {
-	resolver, err := NewResolver(mise.TestInstallDir)
-	require.NoError(t, err)
-
-	resolver.Default("python", "3.11")
-	pkg := resolver.Get("python")
-	assert.Equal(t, "3.11", pkg.Version)
-	assert.Equal(t, DefaultSource, pkg.Source)
-
-	resolvedPackages, err := resolver.ResolvePackages()
-	require.NoError(t, err)
-	assert.Equal(t, 1, len(resolvedPackages))
-}