Fix worker class name for mock