Removed importing types

had removed it's usage, but forgot to remove the import.
Fixed setup

tests is a package so it could be called from withing setup scripts.