corrected  one bug from last commit