Fixed typo - used wrong class