small update to deal with edge case