diff --git a/dgad/models/robust.h5 b/dgad/models/robust.h5
new file mode 100644
index 0000000000000000000000000000000000000000..bfd324acb9aab7cbca2fe83c412cb4bcf3d8d28e
Binary files /dev/null and b/dgad/models/robust.h5 differ
diff --git a/dgad/prediction.py b/dgad/prediction.py
index e4ea217757128ed2984b300787917dabedb01b45..f8f613a2f791de5f84c86398a858556e729275e9 100644
--- a/dgad/prediction.py
+++ b/dgad/prediction.py
@@ -50,7 +50,7 @@ class Detective:
         if model_binary:
             self.model_binary = model_binary
         else:
-            with resources.path(dgad.models, "tcn_best.h5") as model_path:
+            with resources.path(dgad.models, "robust.h5") as model_path:
                 self.model_binary = Model(
                     filepath=model_path, custom_objects={"TCN": TCN}
                 )