fixing bug from last commit