Add unit tests.

Remove this line and called flag.