Removing unnecessary test codes