Fix the model attachment to data source

/to @ritch