Add logging impl for testing
fix broken test