#655 - Fix broken test