diff --git a/CHANGELOG.md b/CHANGELOG.md
index 3fb402c3..ce0d49a7 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -14,6 +14,10 @@ The following table shows which versions of MapReader are compatible with which
 ## Pre-release
 _Add new changes here_
 
+### Fixed
+
+- Fixes the `model_summary` method in the `ClassifierContainer` class ([#574](https://github.com/maps-as-data/MapReader/pull/574))
+
 ## [v1.8.2](https://github.com/Living-with-machines/MapReader/releases/tag/v1.8.2) (2025-12-19)
 
 ### Added
diff --git a/mapreader/classify/classifier.py b/mapreader/classify/classifier.py
index 671d41bd..46513edf 100644
--- a/mapreader/classify/classifier.py
+++ b/mapreader/classify/classifier.py
@@ -565,7 +565,7 @@ def model_summary(
             col_names = ["output_size", "output_size", "num_params"]
 
         model_summary = summary(
-            self.model, input_size=input_size, col_names=col_names, **kwargs
+            self.model, input_size=input_size, col_names=col_names, device=self.device, **kwargs
         )
         print(model_summary)