fix integration with part b