diff --git a/torch_vggish.py b/torch_vggish.py
index f1b67a3..ad85496 100644
--- a/torch_vggish.py
+++ b/torch_vggish.py
@@ -42,6 +42,6 @@ class TorchVggish(Operator):
 
     def __call__(self, audio_path: str) -> NamedTuple('Outputs', [('embs', numpy.ndarray)]):
         audio_tensors = self.model.preprocess(audio_path)
-        features = self.model._model(audio_tensors)
+        features = self.model(audio_tensors)
         Outputs = NamedTuple('Outputs', [('embs', numpy.ndarray)])
         return Outputs(features.detach().numpy())