fixed bug from previous change
Merge pull request #6 from Bharat311/f-remove-deprecation-warning

fixed bug from previous change