Removed import left from tests