Changed it back. Previously the code depended on a method in the class, but now not anymore.