Forgot to remove the assert