Fixing nirvana test - missing mock