fixing bug in last commit