Fix unresolved class