Fix bug: incorrectly sized map