Fixed bug I added with last commit