Putting this logic here for now, it'll probably move.