readd unit test that got lost in a merge