Fix a broken include