Better exception handling in partition split