fixing the integration test failure in ws module.