fixing bug w/ updating