Drag button fix