Fixes broken mock