Correctly transform assert statements