diff --git a/pkg/cli/infra/init.go b/pkg/cli/infra/init.go
index 31b253b4..9242c1a3 100644
--- a/pkg/cli/infra/init.go
+++ b/pkg/cli/infra/init.go
@@ -296,6 +296,8 @@ func getEditorCommand() string {
 		ret = "atom -w"
 	case "subl":
 		ret = "subl -n -w"
+	case "code":
+		ret = "code -n -w"
 	case "mate":
 		ret = "mate -w"
 	case "vim":