Fix failing test in ExtractTest