[ecs] Everywhere `QueryExtension` is used, check result for null

[ikkentim]

Theoretically, QueryExtension can return null. need to check the results where the method is used and handle it appropriately.