Add in tests and get rid of an outdated comment.