Small refactor of callback to use node convention