Add assert statements back