Fix callback in forge