Fix map bug