Fixed: wrong class