diff --git a/colab/widgets/tests/test_widgets.py b/colab/widgets/tests/test_widgets.py
index 824f99a..9f913aa 100644
--- a/colab/widgets/tests/test_widgets.py
+++ b/colab/widgets/tests/test_widgets.py
@@ -2,16 +2,22 @@ import unittest
 from mock import patch
 
 from colab.widgets.templatetags.widgets_tag import import_widgets
-from colab.widgets.widget_manager import WidgetManager
+from colab.widgets.widget_manager import WidgetManager, Widget
+from django.template import Context
+
+
+class WigetMock(Widget):
+    def __init__(self, content=""):
+        self.content = content
 
 
 class WidgetsTest(unittest.TestCase):
     @patch.object(WidgetManager, 'get_widgets')
     def test_import_widgets_tag(self, get_widgets):
-        return_list = [1, 2, 3]
+        return_list = [WigetMock(), WigetMock(), WigetMock()]
         get_widgets.return_value = return_list
 
-        context = {'request': ""}
+        context = Context({'request': ""})
         import_widgets(context, 'area')
 
         self.assertIn('widgets_area', context)
@@ -19,10 +25,10 @@ class WidgetsTest(unittest.TestCase):
 
     @patch.object(WidgetManager, 'get_widgets')
     def test_import_widgets_tag_with_named_var(self, get_widgets):
-        return_list = [1, 2, 3]
+        return_list = [WigetMock(), WigetMock(), WigetMock()]
         get_widgets.return_value = return_list
 
-        context = {'request': ""}
+        context = Context({'request': ""})
         import_widgets(context, 'area', 'var')
 
         self.assertIn('var', context)
--
libgit2 0.21.2