Small fix to single partition functionality