Missed some test updates