Fix broken specs due to incorrect setup