Fixed an issue with the recent change to a class.