Remove some testing code