BUG: small fix