Fix Daemon tests