Remove "ugly hack"