Forgot to add this as part of last commit (for tests).