Reverted the changes made for the test