Fix mock class naming mismatch