Minor fix for testing.