am c7cd2cb2: Add callback which got missed in refactor

* commit 'c7cd2cb28f0d08dd47c79007794fc37036210b6a':
  Add callback which got missed in refactor