prediction tests fix