Removed unneeded mock class