diff --git a/test/calibration_script.jl b/test/calibration_script.jl
index ddd4f6f..8c93d36 100644
--- a/test/calibration_script.jl
+++ b/test/calibration_script.jl
@@ -1,14 +1,12 @@
-using BeforeIT, MAT
-using FileIO, Dates
+
+using MAT, FileIO, Dates
 
 dir = @__DIR__
 
 parameters_mat = matread(joinpath(dir, "matlab_code/italy_calibration/parameters/2010Q1.mat"))
 initial_conditions_mat = matread(joinpath(dir, "matlab_code/italy_calibration/initial_conditions/2010Q1.mat"))
 
-# parameters = BeforeIT.ITALY2010Q1.parameters
-# initial_conditions = BeforeIT.ITALY2010Q1.initial_conditions
-
+using BeforeIT
 
 calibration_data = BeforeIT.ITALY_CALIBRATION.calibration
 figaro = BeforeIT.ITALY_CALIBRATION.figaro
@@ -23,13 +21,12 @@ estimation_date = DateTime(1996, 12, 31)
 # Calibrate on a specific quarter
 calibration_date = DateTime(2010, 03, 31)#-Dates.Month(3)
 parameters, initial_conditions = BeforeIT.get_params_and_initial_conditions(
-    calibration_date,
-    calibration_data,
-    figaro,
-    data,
-    ea,
-    max_calibration_date,
-    estimation_date;
+    (calibration = calibration_data,
+    figaro = figaro,
+    data = data,
+    ea = ea,
+    max_calibration_date = max_calibration_date,
+    estimation_date = estimation_date), calibration_date;
     scale = 0.001,
 )