diff --git a/im2deep/__init__.py b/im2deep/__init__.py
index d1be004..c7e6b2e 100644
--- a/im2deep/__init__.py
+++ b/im2deep/__init__.py
@@ -1,3 +1,3 @@
 """IM2Deep: Deep learning framework for peptide collisional cross section prediction."""
 
-__version__ = "0.1.9"
+__version__ = "0.2.0"
diff --git a/im2deep/im2deep.py b/im2deep/im2deep.py
index 9363144..7bd43df 100644
--- a/im2deep/im2deep.py
+++ b/im2deep/im2deep.py
@@ -31,7 +31,7 @@ def predict_ccs(
     if model_name == "tims":
         path_model = Path(__file__).parent / "models" / "TIMS"
 
-    path_model_list = list(path_model.glob("*.hdf5"))
+    path_model_list = list(path_model.glob("*.keras"))
     if use_single_model:
         LOGGER.debug("Using model {}".format(path_model_list[2]))
         path_model_list = [path_model_list[2]]
diff --git a/im2deep/models/TIMS/full_hc_peprec_CCS_v4_1fd8363d9af9dcad3be7553c39396960.hdf5 b/im2deep/models/TIMS/full_hc_peprec_CCS_v4_1fd8363d9af9dcad3be7553c39396960.keras
similarity index 64%
rename from im2deep/models/TIMS/full_hc_peprec_CCS_v4_1fd8363d9af9dcad3be7553c39396960.hdf5
rename to im2deep/models/TIMS/full_hc_peprec_CCS_v4_1fd8363d9af9dcad3be7553c39396960.keras
index 083c28d..61ba2b9 100644
Binary files a/im2deep/models/TIMS/full_hc_peprec_CCS_v4_1fd8363d9af9dcad3be7553c39396960.hdf5 and b/im2deep/models/TIMS/full_hc_peprec_CCS_v4_1fd8363d9af9dcad3be7553c39396960.keras differ
diff --git a/im2deep/models/TIMS/full_hc_peprec_CCS_v4_8c22d89667368f2f02ad996469ba157e.hdf5 b/im2deep/models/TIMS/full_hc_peprec_CCS_v4_8c22d89667368f2f02ad996469ba157e.hdf5
deleted file mode 100644
index 6c19df3..0000000
Binary files a/im2deep/models/TIMS/full_hc_peprec_CCS_v4_8c22d89667368f2f02ad996469ba157e.hdf5 and /dev/null differ
diff --git a/im2deep/models/TIMS/full_hc_peprec_CCS_v4_8c22d89667368f2f02ad996469ba157e.keras b/im2deep/models/TIMS/full_hc_peprec_CCS_v4_8c22d89667368f2f02ad996469ba157e.keras
new file mode 100644
index 0000000..eec433c
Binary files /dev/null and b/im2deep/models/TIMS/full_hc_peprec_CCS_v4_8c22d89667368f2f02ad996469ba157e.keras differ
diff --git a/im2deep/models/TIMS/full_hc_peprec_CCS_v4_cb975cfdd4105f97efa0b3afffe075cc.hdf5 b/im2deep/models/TIMS/full_hc_peprec_CCS_v4_cb975cfdd4105f97efa0b3afffe075cc.keras
similarity index 55%
rename from im2deep/models/TIMS/full_hc_peprec_CCS_v4_cb975cfdd4105f97efa0b3afffe075cc.hdf5
rename to im2deep/models/TIMS/full_hc_peprec_CCS_v4_cb975cfdd4105f97efa0b3afffe075cc.keras
index b592bc3..a0bf23f 100644
Binary files a/im2deep/models/TIMS/full_hc_peprec_CCS_v4_cb975cfdd4105f97efa0b3afffe075cc.hdf5 and b/im2deep/models/TIMS/full_hc_peprec_CCS_v4_cb975cfdd4105f97efa0b3afffe075cc.keras differ
diff --git a/pyproject.toml b/pyproject.toml
index 7e900d8..73541a2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,7 +20,7 @@ dynamic = ["version"]
 requires-python = ">=3.8"
 dependencies = [
     "click",
-    "deeplc==2.2.38",
+    "deeplc",
     "psm_utils",
     "pandas",
     "numpy",