Fixes expected output