Fix error with map