Fix tests broken in e147518