small fix
optimization