Fix messed up previous commit