MODULE 3: Contract correction