Correct crash introduced by previous commit