Quick fix for test.