fix optimize test after function node layout change