fixed unresolved class use