Fix for the failing test

The test was based on previously desired assumption that is no longer
correct.