Fixed simple callback case

Complex node wrap case still broken.