minor fix
small fix/optimization