Improve set_code_fail/floppy_ret/disk_ret code generation.