Fix broken method