small fix allocation