Fixes on the assegai module wrapper.