fix a problem in code