Minor callback fix