Add logging to test