Removed unnecessary mock