Fixed exception in code issue.