fix broke tests