Hopefully fixed the MLP