Fixed failing test's expectations