After using Return while KO'd you can engage the enemy and going into battle mode.
This cancles your return right after KO.
Game roughly gives you about 5secs after selecting return after a KO to do anything before the return loading is actually done.