Update to fix the class cast exception.