Revert "put prediction into try to be sure"

This reverts commit 9a9ac1ba4fefb5b1c199db76425cbea3d427d302.