minor fix on model