I kept the assertion at line 44, which relies on lines 19-20 in setup.