Fix reference model

- Function signature has changed (line #1797)
- Fix unit test