FIX: Fit the data in tests