Fixed docs based on recent PR