diff --git a/scripts/models.py b/scripts/models.py
index bcb99dd..644696f 100644
--- a/scripts/models.py
+++ b/scripts/models.py
@@ -131,7 +131,7 @@ class PhaseNetLit(SeisBenchModuleLit):
         self.sigma = sigma
         self.sample_boundaries = sample_boundaries
         self.loss = vector_cross_entropy
-        self.model = sbm.PhaseNet(phases="PN", **kwargs)
+        self.model = sbm.PhaseNet(**kwargs)
 
     def forward(self, x):
         return self.model(x)