add assert back in