Fix issue where driver class cannot be used in Python 2.x