Fix docstringt