have to account for both cases