fixing deprecation warning
Merge pull request #18 from jmacglashan/fix_deprecation

fixing deprecation warning