Fixat lite kod som slank med förra committen.