From 13c5ed0bb23c2664902c6b7e2204caa90f4d7cfa Mon Sep 17 00:00:00 2001
From: Per Lindgren <per.lindgren@ltu.se>
Date: Tue, 6 Apr 2021 17:02:24 +0200
Subject: [PATCH] polish

---
 src/lib.rs | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 9b7a039..7ebc3e5 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -138,12 +138,13 @@ where
             Err(Error::Uninitialized)
         } else {
             interrupt_free(|| unsafe {
-                let index = *self.head.get_mut_unchecked();
-                *self.head.get_mut_unchecked() =
-                    *self.free.get_mut_unchecked().get_unchecked(index);
-                if *self.head.get_mut_unchecked() > S {
+                if *self.head.get_mut_unchecked() >= S {
                     Err(Error::OutOfMemory)
                 } else {
+                    let index = *self.head.get_mut_unchecked();
+                    *self.head.get_mut_unchecked() =
+                        *self.free.get_mut_unchecked().get_unchecked(index);
+
                     Ok(Box {
                         index,
                         allocator: self,
-- 
GitLab