Fix lint error in tests