Remove unnecessary mock