Is this change also required to make the test work?