[Bugfix] Forgot to adjust test