Atempting to fix mockery error