Remove innecesary code in test