another little fix to match the model