Checking update code works