Minor go button fix