Added a test for restart or return.