Bugfixes prediction appearance part

Some minor fixes after the cleanup of variables