Adds missing import
Merge pull request #1 from z4y4ts/patch-1

Adds missing import