Remove leftover code from testing