... testing is hard. Fixing the test...