Fix broken test due to a bad rename