Mark method return as NonNull

The interface has changed in a recent commit.