Quick fix for test