Add proper callback for extension