diff --git a/tests/data/prediction/popp_line_model.pt b/tests/data/prediction/popp_line_model.pt
index 67b9565aee8a50f9d9b73fed2deb7e452d3828fb..90d12ddd0354414f2a6eb568a10a967a88e7f49f 100644
Binary files a/tests/data/prediction/popp_line_model.pt and b/tests/data/prediction/popp_line_model.pt differ
diff --git a/tests/data/training/models/best_0.pt b/tests/data/training/models/best_0.pt
index 9190c418d96dac5aafbaa8d466314ef305ed46e7..f26eaa9f5da3793efb788b6202494cd5bc54cf34 100644
Binary files a/tests/data/training/models/best_0.pt and b/tests/data/training/models/best_0.pt differ
diff --git a/tests/data/training/models/last_3.pt b/tests/data/training/models/last_3.pt
index 5c82fb729e19c5e55686b0e675a1a3beec4c7071..6ea2ad1fc7c0bc6d00e355e4f596c11754f92f98 100644
Binary files a/tests/data/training/models/last_3.pt and b/tests/data/training/models/last_3.pt differ