forgot to add these two lines to the last commit