Fixed failing test