Update schema, fixes #320