This fixes it :)
Update docs