Minor fix
Add a mock object for tests