Improved the test now.