Add back the time limit to the go implementation.