Adaptations for use of prediction class.