Fix minor deprecation warning
Merge pull request #126 from petergoldstein/feature/fix_deprecation_warning

Fix minor deprecation warning