Remove space issues in the test code