Reverted changes to test case