Test helpers fixes