Fixes test break.