Fix test.
Make it a little bit faster.