diff --git a/lapce-data/src/data.rs b/lapce-data/src/data.rs
index d9d87b2c..47e88336 100644
--- a/lapce-data/src/data.rs
+++ b/lapce-data/src/data.rs
@@ -25,7 +25,6 @@
     command::{FocusCommand, MultiSelectionCommand},
     cursor::{Cursor, CursorMode},
     editor::EditType,
-    language::LapceLanguage,
     mode::MotionMode,
     movement::Movement,
     register::Register,
@@ -2743,7 +2742,10 @@ pub fn export_theme(&mut self, ctx: &mut EventCtx, config: &Config) {
         let id = self.new_file(ctx, config);
         let doc = self.scratch_docs.get_mut(&id).unwrap();
         let doc = Arc::make_mut(doc);
-        doc.set_language(LapceLanguage::Toml);
+
+        #[cfg(feature = "lang-toml")]
+        doc.set_language(lapce_core::language::LapceLanguage::Toml);
+
         doc.reload(Rope::from(config.export_theme()), true);
     }