fix imports

Signed-off-by: Jaka Hudoklin <jakahudoklin@gmail.com>
Merge pull request #30 from ariskemper/patch-1

fix imports