extend test case to show the return bug