fixing bug from last commit

fixing bug from last commit