Fix code from different issue.