Fix in callback functionality.