tests: Add missing fix from previous commit