small fix/update