Fixed; did a similar thing to the other function.