Added button fix.