fixed the issue from my last commit