Small fixed on GUI