added correct import statment
Merge pull request #3 from rweichler/master

added correct import statment