Updated consumer unit test