diff --git a/README.md b/README.md
index a056a0a..4302c8c 100644
--- a/README.md
+++ b/README.md
@@ -59,7 +59,7 @@ class MNISTTrainer(ETRunner):
 
     def new_meter(self):
         return ETMeter(
-            cfm=ConfusionMatrix(num_classes=10)
+            cfm=ConfusionMatrix(num_classes=10, device=self.device['gpu'])
         )
 
 
diff --git a/setup.py b/setup.py
index 88c1489..e362855 100644
--- a/setup.py
+++ b/setup.py
@@ -29,7 +29,7 @@
 # This call to setup() does all the work
 setup(
     name="easytorch",
-    version="3.8.3",
+    version="3.8.4",
     description="Easy Neural Network Experiments with pytorch",
     long_description=_README,
     long_description_content_type="text/markdown",