'schema_module' seems like an unusual way to import this, 'schema' can be used here as it is not defined anywhere else.