Add logging to test assembly