Fix broken schema

This got broken during a rebase :/