modify testing code