removed an import
minor changes (I know, very bad commit)