Fix argument error.

CLoses #312.