Properly reset drag styles on drag end