Fixed bug that I introduced