Update tests to reproduce bug