for reference, this is the bit i changed to get the test working