fixed broken build for mock version