Fix failing tests

Forgot to change the tests for ontology.py