Update __init__.py

Fix search() to handle arbitrary queries not returning a whole object (where entity_zero is None)