Have updated the test cases as well w.r.t module changed