Fast fix for test