Fix lint errors in tests