Caught a forgotten case on a test and improved it.