Fixed call to class before it existed