diff --git a/tests/test_tree_building.py b/tests/test_tree_building.py
index dc683e0..32874ed 100644
--- a/tests/test_tree_building.py
+++ b/tests/test_tree_building.py
@@ -7,9 +7,7 @@
 
 YASU_TREE = """{}{  |--headers=[ config, pkgs, ... ]}{  |--imports=[  ./hardware-configuration.nix ]}{  boot}{    loader}{      systemd-boot}{        |--enable=true}{      efi}{        |--canTouchEfiVariables=true}{  networking}{    |--hostName='nixos'}{    |--defaultGateway='10.11.12.1'}{    |--nameservers=[ '10.11.12.1' ]}{    firewall}{      |--allowedTCPPorts=[ 3389 ]}{  time}{    |--timeZone='Japan'}{  virtualisation}{    virtualbox}{      host}{        |--enable=true}{  hardware}{    bluetooth}{      |--enable=true}{      config}{        General}{          |--Enable='Source,Sink,Media,Socket'}{    pulseaudio}{      |--enable=true}{      |--extraModules=[ pkgs.pulseaudio-modules-bt ]}{      |--package=pkgs.pulseaudioFull}{      |--support32Bit=true}{      |--extraConfig='' load-module module-bluetooth-policy auto_switch = 2 ''}{    opengl}{      |--driSupport32Bit=true}{  services}{    blueman}{      |--enable=true}{    cron}{      |--enable=true}{      |--systemCronJobs=[ '@reboot root ${pkgs.ethtool}/sbin/ethtool -s enp4s0 wol g' ]}{    openssh}{      |--enable=true}{      |--passwordAuthentication=false}{      |--challengeResponseAuthentication=false}{      |--extraConfig='UseDNS yes'}{    vsftpd}{      |--enable=true}{      |--localUsers=true}{      |--writeEnable=true}{      |--extraConfig='' pasv_enable = YES connect_from_port_20 = YES pasv_min_port = 4242 pasv_max_port = 4243 ''}{    apcupsd}{      |--enable=true}{      |--configText=''  UPSCABLE smart UPSTYPE apcsmart DEVICE /dev/ttyS0 ''}{    postfix}{      |--enable=true}{      |--setSendmail=true}{    xserver}{      |--enable=true}{      |--layout='us'}{      displayManager}{        gdm}{          |--enable=true}{      desktopManager}{        gnome3}{          |--enable=true}{      |--videoDrivers=[ 'nvidia' ]}{    fail2ban}{      |--enable=true}{    netdata}{      |--enable=true}{    xrdp}{      |--enable=true}{      |--defaultWindowManager='${pkgs.icewm}/bin/icewm'}{    vnstat}{      |--enable=true}{  sound}{    |--enable=true}{  programs}{    mosh}{      |--enable=true}{    gnupg}{      agent}{        |--enable=true}{  systemd}{    targets}{      sleep}{        |--enable=false}{      suspend}{        |--enable=false}{      hibernate}{        |--enable=false}{      hybrid-sleep}{        |--enable=false}{  users}{    extraUsers}{      yasu}{        |--home='/home/yasu'}{        |--isNormalUser=true}{        |--uid=1000}{        |--extraGroups=[ 'wheel' ]}{  nixpkgs}{    config}{      |--allowUnfree=true}{  powerManagement}{    |--enable=true}{  i18n}{    inputMethod}{      |--enabled='ibus'}{      ibus}{        |--engines=[ (pkgs.ibus-engines).mozc ]}{  fonts}{    |--fonts=[ (pkgs).carlito (pkgs).dejavu_fonts (pkgs).ipafont (pkgs).kochi-substitute (pkgs).source-code-pro (pkgs).ttf_bitstream_vera ]}{    fontconfig}{      defaultFonts}{        |--monospace=[ 'DejaVu Sans Mono' 'IPAGothic' ]}{        |--sansSerif=[ 'DejaVu Sans' 'IPAPGothic' ]}{        |--serif=[ 'DejaVu Serif' 'IPAPMincho' ]}"""
 
-SHORTENED_DEFAULT_TREE = """{}{  |--headers=[ config, pkgs, ... ]}{  |--imports=[ ./hardware-configuration.nix ]}{  boot}{    loader}{      grub}{        |--enable=true}{        |--device='/dev/sda'}{        |--useOSProber=true}{  networking}{    |--hostName='nixos'}{    networkmanager}{      |--enable=true}{  time}{    |--timeZone='Europe/London'}{  i18n}{    |--defaultLocale='en_GB.UTF-8'}{  services}{    xserver}{      |--enable=true}{  programs}{    firefox}{      |--enable=true}{  nixpkgs}{    config}{      |--allowUnfree=true}{  environment}{    |--systemPackages=[ (pkgs).vim (pkgs).git ]}{  system}{    |--stateVersion='23.11'}"
-
-  - {}{  |--headers=[ config, pkgs, ... ]}{  |--imports=[ ./hardware-configuration.nix ]}{  boot}{    loader}{      grub}{        |--enable=true}{        |--device='/dev/sda'}{        |--useOSProber=true}{  networking}{    |--hostName='nixos'}{    networkmanager}{      |--enable=true}{  time}{    |--timeZone='Europe/London'}{  i18n}{    |--defaultLocale='en_GB.UTF-8'}{  services}{    xserver}{      |--enable=true}{  programs}{    firefox}{      |--enable=true}{  nixpkgs}{    config}{      |--allowUnfree=true}{  environment}{    |--systemPackages=[ (pkgs).vim (pkgs).git ]}{  system}{    |--stateVersion='23.11'}"""
+SHORTENED_DEFAULT_TREE = """{}{  |--headers=[ config, pkgs, ... ]}{  |--imports=[ ./hardware-configuration.nix ]}{  boot}{    loader}{      grub}{        |--enable=true}{        |--device='/dev/sda'}{        |--useOSProber=true}{  networking}{    |--hostName='nixos'}{    networkmanager}{      |--enable=true}{  time}{    |--timeZone='Europe/London'}{  i18n}{    |--defaultLocale='en_GB.UTF-8'}{  services}{    xserver}{      |--enable=true}{  programs}{    firefox}{      |--enable=true}{  nixpkgs}{    config}{      |--allowUnfree=true}{  environment}{    |--systemPackages=[ (pkgs).vim (pkgs).git ]}{  system}{    |--stateVersion='23.11'}"""
 
 def tree_output(node: Node, append: str = "", output_string: str = "") -> str:
     """