Fixed a return value change in a test