Added unit test that was left out from the last commit.