Changed dti predict so that the test would work.