Update check_schema_keys for new DB mappings