Fixing bug where args were dropped when passed to module function