Removed an import

Really minor change