Re-added a missing line
the I accidentally removed... oops :p