diff --git a/run_with_finetuning.py b/run_with_finetuning.py
index 65611793a361717add42155ed914bfc7c7e71321..fc334a5cf111f3343ebf7d87ef7b8d4746febdf8 100644
--- a/run_with_finetuning.py
+++ b/run_with_finetuning.py
@@ -5,7 +5,8 @@ import argparse
 from mlair.workflows import DefaultWorkflow
 from mlair.helpers import remove_items
 from mlair.configuration.defaults import DEFAULT_PLOT_LIST
-from mlair.model_modules.model_class import IntelliO3_ts_architecture, IntelliO3_ts_architecture_freeze
+from mlair.model_modules.model_class import IntelliO3_ts_architecture, IntelliO3_ts_architecture_finetune_all_dense, \
+    IntelliO3_ts_architecture_finetune_outputs, IntelliO3_ts_architecture_finetune_main_output
 import os
 
 
@@ -27,7 +28,7 @@ def main(parser_args):
         #stations=["DEBW013", "DEBW087"],
         epochs=1, external_weights="/home/vincentgramlich/mlair/data/weights/testrun_network_daily_model-best.h5",
         train_model=True, create_new_model=True, network="UBA",
-        model=IntelliO3_ts_architecture_freeze,
+        model=IntelliO3_ts_architecture_finetune_all_dense,
         window_lead_time=1,
         #oversampling_method="bin_oversampling", oversampling_bins=10, oversampling_rates_cap=100, window_lead_time=2,
         evaluate_bootstraps=False,  plot_list=["PlotContingency"],