fixed notify function to work with change in wrapper method