Fix max_tokens == 1 case to match documentation. Add tests.