fix issue with broken test