diff --git a/tests/test_features.py b/tests/test_features.py
index <HASH>..<HASH> 100644
--- a/tests/test_features.py
+++ b/tests/test_features.py
@@ -10,8 +10,9 @@ from __future__ import (absolute_import, division, print_function)
 
 import os
 
+from six import text_type
+
 from branca.element import Element
-from branca.six import text_type
 
 from folium import Map, Popup
 import folium
Fix tests by using regular six instead of branca.six (#<I>)

Use regular six package instead of branca