Minor fix... forgot to delete something