fix mlp bug