Fix error where it was not correctly writing the return of the function in the map
Merge pull request #8 from ftlabs/bugfixes

Fix error where it was not correctly writing the return of the function in the map.