fixed test removed the unwanted code