Changed VirtualHostAwareCodeGenerator to comply with the new generated API.