actually need to update here too