Remove unused mock class