minor fix in mock