small fix 2