Fix rule based on assumption, remove extra use-case, add JSDoc