Another fix for test