Fix tests failing after 4fccfee3