Fix for partition