Fixes GUI bug