Support sub-models for return value casting too