fix class lookup