Fix for breaking test