Quick fix to bring the test back to normal