Fix lint errors in the tests