Fix test to match Dmitry's bug fix