Fixed wrong class