fixed up code snippets after submod change