one more fix to predictions code