Forgot to add this to the previous commit.