diff --git a/src/scripts/lib/tools/MMSplice.py b/src/scripts/lib/tools/MMSplice.py
index d9cf835..9ae5bd8 100644
--- a/src/scripts/lib/tools/MMSplice.py
+++ b/src/scripts/lib/tools/MMSplice.py
@@ -133,14 +133,14 @@ def predict_table_fast(model,
 
 args = parser.parse_args()
 
-dl = SplicingVCFDataloader(args.gtf,
-                           args.fasta,
-                           args.input)
+try:
+    dl = SplicingVCFDataloader(args.gtf,
+                               args.fasta,
+                               args.input)
 
-# Specify model
-model = MMSplice()
+    # Specify model
+    model = MMSplice()
 
-try:
     # Do prediction
     predictions = predict_table_fast(model, dl, batch_size=512)