fixing breaking tests