Made tests work with new custom document and component