Fix UnboundLocalError on enable_world