minor updates for mock testing