Fix a bug in DNNX's solution