Slight fix to get rid of "raw types" warnings
Merge pull request #125 from cogmission/slightfix

Slight fix to get rid of "raw types" warnings