Forgot to modify one line of the old code