DEBUG: update code to check for correction