test: Fix broken test