You'll  want to bring in six and use, I think, `six.string_types`