Fix test import warning
Merge pull request #22 from geopet/test-warning-fix

Fix test import warning