Fixed a forgotten line