Sorted out the test issue.