fix map function