return_address: Fix TODO that never got done

Signed-off-by: Paul Beeler <pbeeler80@gmail.com>