Fix broken test. (justinvdm)