Apply suggestions from code review
Apply suggestions from code review

used the wrong method in a test

Co-Authored-By: Olle Jonsson <bd80a69ef196e05ef31520b7e45ef62ed607aa2a@gmail.com>
jhawthorn/discard,jhawthorn/discard