Fixed the assert statements