Fixed an issue where the wrong class was being used.