minor fixes
fixed unhandled bug