GUI bug fixes