Updating docs, using correct type