Fix compile error due to model component changes