diff --git a/entc/gen/feature.go b/entc/gen/feature.go
index 6626913da..a75fbc135 100644
--- a/entc/gen/feature.go
+++ b/entc/gen/feature.go
@@ -39,6 +39,12 @@ var (
 		Stage:       Experimental,
 		Default:     false,
 		Description: "Schema snapshot stores a snapshot of ent/schema and auto-solve merge-conflict (issue #852)",
+		GraphTemplates: []GraphTemplate{
+			{
+				Name:   "internal/schema",
+				Format: "internal/schema.go",
+			},
+		},
 		cleanup: func(c *Config) error {
 			return remove(filepath.Join(c.Target, "internal"), "schema.go")
 		},
diff --git a/entc/gen/template.go b/entc/gen/template.go
index 66787b151..f19d2e7c1 100644
--- a/entc/gen/template.go
+++ b/entc/gen/template.go
@@ -155,13 +155,6 @@ var (
 			Name:   "runtime/pkg",
 			Format: "runtime/runtime.go",
 		},
-		{
-			Name:   "internal/schema",
-			Format: "internal/schema.go",
-			Skip: func(g *Graph) bool {
-				return !g.featureEnabled(FeatureSnapshot)
-			},
-		},
 	}
 	// patterns for extending partial-templates (included by other templates).
 	partialPatterns = [...]string{