Small fix for drag-n-drop