Minor fix for testing