Fixed the assembly reference issue.