removed unushed import
returning correct type