We'll need to update this to be more generic.