Fix CNN back to original