Fixed GUI bug