Restore bug fix, make previously added test pass.

1 job for 315135-non-regression-test in 5 minutes and 48 seconds