Fix error on schema

This fixes the issue I reported (#1)