Do we want to do any unit tests here for these methods? Or are we :ok_hand: with checking `display_type`.