[#362] Fix broken test code