fix new code