diff --git a/gooey/tests/i18n_unittest.py b/gooey/tests/i18n_unittest.py
index 97f298c..6461902 100644
--- a/gooey/tests/i18n_unittest.py
+++ b/gooey/tests/i18n_unittest.py
@@ -16,10 +16,10 @@
 #
 #     i18n.load('english')
 #     self.assertTrue(i18n._DICTIONARY is not None)
-#     self.assertEqual('Cancel', i18n.translate('cancel'))
+#     self.assertEqual('Cancel', i18n._('cancel'))
 #
 #     i18n.load('french')
-#     self.assertEqual('Annuler', i18n.translate('cancel'))
+#     self.assertEqual('Annuler', i18n._('cancel'))
 #
 #
 #   def test_i18n_throws_exception_on_no_lang_file_found(self):