Fix a minor issue with the assert function and make a stylistic change.