Fixed the bug I added in the last commit