Fixed the issue that I was seeing before