Simple map fix bug