Missing include was breaking tests