diff --git a/dace/cli/sdfg_diff.py b/dace/cli/sdfg_diff.py
index 2ec0a3adf4..ed661b8c25 100644
--- a/dace/cli/sdfg_diff.py
+++ b/dace/cli/sdfg_diff.py
@@ -7,7 +7,11 @@
 import os
 import platform
 import tempfile
-from typing import Dict, Literal, Set, Tuple, Union
+from typing import Dict, Set, Tuple, Union
+try:
+    from typing import Literal
+except ImportError:
+    from typing_extensions import Literal
 
 import dace
 from dace import memlet as mlt