Fix nil map error