Small edits for Test