bugfix: forgot to mock a class