fix for breaking tests