Callbacks fix + TODO