fix missing includes breaking tests