Modified the test - made changes in asserting the test