FIX epochs picks

Eric, this fixes the test + cleans up the related example.