take care of FutureWarning
Merge pull request #318 from vnitinv/fix-warning

take care of FutureWarning