@@ -16,10 +16,12 @@
 
 #import "OFStream.h"
 
 #import "socket.h"
 
+OF_ASSUME_NONNULL_BEGIN
+
 /*!
  * @class OFStreamSocket OFStreamSocket.h ObjFW/OFStreamSocket.h
  *
  * @brief A class which provides functions to create and use stream sockets.
  */
@@ -34,5 +36,7 @@
  *
  * @return A new, autoreleased OFTCPSocket
  */
 + (instancetype)socket;
 @end
+
+OF_ASSUME_NONNULL_END