Works with API generation