diff --git a/main.py b/main.py
index 16fe3141aa6669c871986483039a1d4a1f3e9f24..1432310a1c1d9e35f96d4ea1273c29ffafe8f45a 100644
--- a/main.py
+++ b/main.py
@@ -397,7 +397,7 @@ if __name__ == '__main__':
 
 
     # initializes MusicNet, MlP/DifficultyNet, transformer
-    danceformer = Danceformer().to(device)
+    danceformer = Danceformer(indexed_vocabulary).to(device)
     print(danceformer)
     # number parameter
     print("Number parameters:")