Fix lint

fixing UT error