fix deprecated
Merge pull request #5 from yanns/cleanup

fix deprecated