Simplified the test case