Fix lint

fixing UT error

Review Changes