remove unncessary method, fix test