Adding partition check