Commented out the correct function by mistake in the previous commit.
Corrected.