Fixes CI, test should pass