Fix incorrect import.
Merge pull request #278 from greboid/master

Fix incorrect import.