Update epochs.py

cc @agramfort | @Eric89GXL this fixes a stupid bug that broke at least one example.