add missing testing code