Fixed wrong class uses.