Fix broken test from contribution