Fix ImageTree test