fix MLP again