W1 Ex 19: fix test and solution to match assignment

Closes #1