Removing unnecssary code in test file.