Add logging info
fix test