fixed issue where python 2 will cannot import a data class