Fixed bug which I introduced