diff --git a/cvm/utils/utils.py b/cvm/utils/utils.py
index d1b8de7..07a35bb 100644
--- a/cvm/utils/utils.py
+++ b/cvm/utils/utils.py
@@ -184,8 +184,8 @@ def list_models(lib: str = 'all'):
     assert lib in ['all', 'cvm', 'torch', 'timm'], f'Unknown library {lib}.'
 
     if lib == 'all':
-        cvm_models = _filter_models(torchvision.models.__dict__, sort=True)
-        torch_models = _filter_models(models.__dict__, 'torch/', True)
+        cvm_models = _filter_models(models.__dict__, sort=True)
+        torch_models = _filter_models(torchvision.models.__dict__, 'torch/', sort=True)
 
         timm_models = [
             'timm/' + name for name in timm.list_models()
diff --git a/cvm/version.py b/cvm/version.py
index 99c4176..210ebb3 100644
--- a/cvm/version.py
+++ b/cvm/version.py
@@ -1 +1 @@
-__version__ = '0.0.1'
\ No newline at end of file
+__version__ = '0.0.2'
\ No newline at end of file