A small fix to make the test working again