Fix assert statements