Improved the tests for the 'return' action.