Solved a bug that I introduced when testing the previous feature