fix failing unit tests
Merge pull request #521 from swatisingh10/failingTestCases

fix failing unit tests