Doc string fix