removing unnecessary call to compute prediction