Add more logging in test