I think you want to change this "if" condition too.