diff --git a/internal/nbconn/nbconn_test.go b/internal/nbconn/nbconn_test.go
index c4a8e65b..8b672e4e 100644
--- a/internal/nbconn/nbconn_test.go
+++ b/internal/nbconn/nbconn_test.go
@@ -294,7 +294,7 @@ func TestInternalNonBlockingWrite(t *testing.T) {
 		}()
 
 		readBuf := make([]byte, deadlockSize)
-		_, err = conn.Read(readBuf)
+		_, err = io.ReadFull(conn, readBuf)
 		require.NoError(t, err)
 
 		err = conn.Close()