Correct the expectation (thanks @Habbie)