Fixing categorical dtype test