Fix UnboundLocalError