Little test fix