fixed test error and removed unused importd