Remove the some temp code