Fixed one minor thing in the previous commit