Correct partition timeout logic