hot fix, fix broken tests