Quick fix to some tests.