small fix for dropPartition, add a test case