Adjusted test case to be more consistent in assert style.