actually fixing bug