Update breaking tests