Forgot a piece of this pull request