Fixed add_callback input in utils.py