Fix feature map error