Fix bug with prediction visibilty