Made huge diagram test work again