Fixing broken schema code