Fix failing test of calling functions for related models