fix: small cleanup