Fix mock class names in test