Fix update plan bug