Fix for #3331

Fix for #3331
Was just following previous practices in the test, but ok.