Removed redundant code in the test file