Fix ofr the last commit's issue.