Remove the testing code from previous commit