Fix for docs