[Bugfix] Forgot to adjust tests