diff --git a/lang/gen.go b/lang/gen.go
index e4fcd62b..7268acf0 100644
--- a/lang/gen.go
+++ b/lang/gen.go
@@ -444,7 +444,7 @@ func GetSolutionCode(q *leetcode.QuestionData) (string, error) {
 	var codeLinesToKeep []string
 	inCode := false
 	for _, line := range codeLines {
-		if !inCode && strings.Contains(line, config.CodeEndMarker) {
+		if !inCode && strings.Contains(line, config.CodeBeginMarker) {
 			inCode = true
 			continue
 		}