Fix for documentation