Add allocation checks for map usage