Fix test and adding another a regression test.