Modified test_schema custom module file