Fixed #234 Generated tests don't compile (2.0.0-beta3)