Fix test after implementing #635