Fix tests

They were not updated to match 95a862b49a1864a4e32c39fab5cca029574d4026