BUG: Handle scalar case for attach_columns