fix to return_code colorization