fix: small fixes