fix mock in test broken after review fixes