Implement omit for module args, also add tests for it