[Misc] Improve logging and remove stack trace from unit test + improve unit test