fix unit test to match actual behavior