Fix max_col ==1