small fix for one bug