Fix for docs again