Fixed the issue as per the comments