Supplement the testing code