[BUGFIX] change model reference