fix(model): minor fix