Add a proper test

liam morland → made their first commit to this issue’s fork.

As a start, these changes get tests passing again.