Update deprecated import
Merge pull request #1 from ToeKnee/master

Update deprecated import