Fix assert statement