diff --git a/lib/consts.dart b/lib/consts.dart
index 1bdbdc4a..537b040d 100644
--- a/lib/consts.dart
+++ b/lib/consts.dart
@@ -230,11 +230,13 @@ const kDefaultHttpMethod = HTTPVerb.get;
 const kDefaultContentType = ContentType.json;
 
 enum CodegenLanguage {
-  dartHttp("Dart (http)"),
-  kotlinOkHttp("Kotlin (OkHttp)");
+  dartHttp("Dart (http)", "dart", "dart"),
+  kotlinOkHttp("Kotlin (OkHttp)", "java", "kt");
 
-  const CodegenLanguage(this.label);
+  const CodegenLanguage(this.label, this.codeHighlightLang, this.ext);
   final String label;
+  final String codeHighlightLang;
+  final String ext;
 }
 
 const JsonEncoder kEncoder = JsonEncoder.withIndent('  ');
diff --git a/lib/widgets/codegen_previewer.dart b/lib/widgets/codegen_previewer.dart
index 02043805..a4083661 100644
--- a/lib/widgets/codegen_previewer.dart
+++ b/lib/widgets/codegen_previewer.dart
@@ -160,7 +160,7 @@ class _ViewCodePaneState extends State<ViewCodePane> {
                     ),
                     SaveInDownloadsButton(
                       content: stringToBytes(widget.code),
-                      mimeType: "application/vnd.dart",
+                      ext: widget.codegenLanguage.ext,
                       showLabel: showLabel,
                     )
                   ],
@@ -175,7 +175,7 @@ class _ViewCodePaneState extends State<ViewCodePane> {
                   child: CodeGenPreviewer(
                     code: widget.code,
                     theme: codeTheme,
-                    language: 'dart',
+                    language: widget.codegenLanguage.codeHighlightLang,
                     textStyle: kCodeStyle,
                   ),
                 ),