I think I gave it the wrong variable - also, reverted last commit