diff --git a/audio_embedding.py b/audio_embedding.py
index 8e08e1d..eb9cb9c 100644
--- a/audio_embedding.py
+++ b/audio_embedding.py
@@ -31,6 +31,6 @@ def AudioEmbedding(config=None):
     return (
         pipe.input('path')
             .map('path', 'frame', ops.audio_decode.ffmpeg(config.batch_size, config.sample_rate, config.layout))
-            .map('frame', 'vec', ops.audio_embedding.vggish(config.weights_path, config.framework), conf=op_config)
+            .map('frame', 'vec', ops.audio_embedding.vggish(config.weights_path, config.framework), config=op_config)
             .output('vec')
     )