CutCode-org
diff --git a/‎src/CutCode.CrossPlatform/ViewModels/CodeViewModel.cs‎
Lines changed: 16 additions & 1 deletion b/‎src/CutCode.CrossPlatform/ViewModels/CodeViewModel.cs‎
Lines changed: 16 additions & 1 deletion
@@ -18,7 +18,7 @@ public class CodeViewModel : PageBaseViewModel, IRoutableViewModel
 {
     public CodeModel Code;
 
-    public Language _language;
+    public Language? _language;
 
     public CodeViewModel(CodeModel code)
     {
@@ -38,6 +38,8 @@ public void Initialise(CodeModel code)
 
         var reg = new RegistryOptions(ThemeName.Dark);
         _language = reg.GetLanguageByExtension(code.Language);
+        if (_language is null)
+            UpdateToNewLanguages(code.Language, reg);
         Language = _language.ToString();
         IsEditEnabled = false;
 
@@ -69,6 +71,19 @@ public void Initialise(CodeModel code)
         IsFavouritePath = code.IsFavourite ? IconPaths.StarFull : IconPaths.Star;
     }
 
+    private void UpdateToNewLanguages(string language, RegistryOptions reg)
+    {
+        var all = reg.GetAvailableLanguages();
+        _language = language switch
+        {
+            "Python" => reg.GetLanguageByExtension(".py"),
+            "c++" => reg.GetLanguageByExtension(".cpp"),
+            "C#" => reg.GetLanguageByExtension(".cs"),
+            "CSS" => reg.GetLanguageByExtension(".css"),
+            _ => _language
+        };
+    }
+
     public ObservableCollection<CodeCellViewModel?> Cells { get; set; }
 
     [Reactive] public string Title { get; set; }