fix failing test

Does this represent the intent of the original code?