fix bug with partition columns not set