Fix/map bugs