Fixes test breaker