restored mlp_hook