diff --git a/sbb_binarize/sbb_binarize.py b/sbb_binarize/sbb_binarize.py
index 8960354..247d54b 100644
--- a/sbb_binarize/sbb_binarize.py
+++ b/sbb_binarize/sbb_binarize.py
@@ -34,6 +34,8 @@ class SbbBinarizer:
         self.start_new_session()
 
         self.model_files = glob('%s/*.h5' % self.model_dir)
+        if not self.model_files:
+            self.model_files = glob('%s/*/' % self.model_dir)
         if not self.model_files:
             raise ValueError(f"No models found in {self.model_dir}")