Fix page flow issue in partition module.