Minor change to fix docs