Fix deprication warning
Merge pull request #4 from InfoTech/fix-deprication

Fix deprication warning