Fixed up test logic but we want to be on another node level