Remove testing code from previous commit