I feel like it would be better to do this in update().