Fix consumer tests