fix use statement and wrong class name