Comment button fix