Add test of return value on stage 2