diff --git a/mace/__version__.py b/mace/__version__.py
index a8d4557d..d7b30e12 100644
--- a/mace/__version__.py
+++ b/mace/__version__.py
@@ -1 +1 @@
-__version__ = "0.3.5"
+__version__ = "0.3.6"
diff --git a/mace/calculators/foundations_models.py b/mace/calculators/foundations_models.py
index 1ee7659a..bc9ed654 100644
--- a/mace/calculators/foundations_models.py
+++ b/mace/calculators/foundations_models.py
@@ -1,7 +1,7 @@
 import os
 import urllib.request
 from pathlib import Path
-from typing import Literal, Union
+from typing import Union
 
 import torch
 from ase import units
@@ -20,7 +20,7 @@ def mace_mp(
     device: str = "",
     default_dtype: str = "float32",
     dispersion: bool = False,
-    damping: Literal["zero", "bj", "zerom", "bjm"] = "bj",
+    damping: str = "bj",  # choices: ["zero", "bj", "zerom", "bjm"]
     dispersion_xc: str = "pbe",
     dispersion_cutoff: float = 40.0 * units.Bohr,
     **kwargs,