diff --git a/rollbar/test/test_shortener_transform.py b/rollbar/test/test_shortener_transform.py
index 3bcc458c..b1c3f796 100644
--- a/rollbar/test/test_shortener_transform.py
+++ b/rollbar/test/test_shortener_transform.py
@@ -47,7 +47,7 @@ def _assert_shortened(self, key, expected):
         result = transforms.transform(self.data, shortener)
 
         if key == 'dict':
-            self.assertEqual(expected, len(result))
+            self.assertEqual(expected, len(result[key]))
         elif key in ('list_max_level', 'dict_max_level'):
             self.assertEqual(expected,  result[key])
         else: