Changed assertion code