Fix premature commit and merge where I was missing a line, sorry!!