Quick thing I forgot to add in the last commit