Fix docstring

Change-Id: Ie737cbb7f8b1fa8a79bbb88914730b05aa7f2906