Fixed security issue for prediction