forgot to edit tests when fixing bug