Bugfix: display prob