Lab7 test fix