Finally some unit tests...