[#627] Fix test broken due to default schema change
PEP8