Fix bug with callback