fix UnboundLocalError