Small changes to test file