[Bugfix] Forgot to adjust some tests