Fix optimization logger callback registration (#28)