Fix UnboundLocalError: local variable 'e'