Minor chages to test