Fix failing test due to altered model