Fix partition return value, should be an array