Fix map test