Fixing ImportError
Using text_type instead of string_types