Fix a bug in marginal logregr