Added fix for docs