This was the line that I needed to change to get the test to work.