Added unit test that was missed in last commit