Fixed enum not found errors
Merge pull request #2 from CivilizedGravy/master

Fixed enum not found errors