fix new code bug