Fix mock data for unit tests