Added some tests.
Tiny fix.