Fix test for event generator