fix partition tests