Fix for broken docs