diff --git a/README.md b/README.md
index 561285b..ea04b80 100644
--- a/README.md
+++ b/README.md
@@ -13,7 +13,7 @@ pip3 install torch==1.8.1+cu111  torchaudio==0.8.1 -f https://download.pytorch.o
 # Installation 
 Install the latest version of this repo:
 ```shell
-pip install -e 'git+https://github.com/kkoutini/passt_hear21@0.0.22#egg=hear21passt' 
+pip install -e 'git+https://github.com/kkoutini/passt_hear21@0.0.23#egg=hear21passt' 
 ```
 
 The models follow the [common API](https://neuralaudio.ai/hear2021-holistic-evaluation-of-audio-representations.html#common-api) of HEAR 21 
diff --git a/hear21passt/__init__.py b/hear21passt/__init__.py
index 4df4962..eb4e26b 100644
--- a/hear21passt/__init__.py
+++ b/hear21passt/__init__.py
@@ -1,5 +1,5 @@
 
-__version__ = "0.0.22"
+__version__ = "0.0.23"
 
 
 def embeding_size(hop=50, embeding_size=1000):
diff --git a/hear21passt/wrapper.py b/hear21passt/wrapper.py
index d731168..c909884 100644
--- a/hear21passt/wrapper.py
+++ b/hear21passt/wrapper.py
@@ -5,7 +5,7 @@
 
 class PasstBasicWrapper(nn.Module):
     def __init__(self, mel: nn.Module, net: nn.Module, max_model_window=10000, timestamp_window=160, timestamp_hop=50,
-                 scene_hop=2500, scene_embedding_size=1295, timestamp_embedding_size=1295, mode="all"):
+                 scene_hop=2500, scene_embedding_size=1295, timestamp_embedding_size=1295, mode="all", **kwargs):
         """
         @param mel: spectrogram extractor
         @param net: network module