disclaimer: I just pushed buttons and this worked (fixed my test cases). I actually have no idea what I'm doing.