Added unit test which got lost in the last commit.