Fix argument type in initWithDataModel