diff --git a/src/algorithms/math/integer-partition/__test__/integerPartition.test.js b/src/algorithms/math/integer-partition/__test__/integerPartition.test.js
index affcbca6..e78b1dfd 100644
--- a/src/algorithms/math/integer-partition/__test__/integerPartition.test.js
+++ b/src/algorithms/math/integer-partition/__test__/integerPartition.test.js
@@ -6,6 +6,9 @@ describe('integerPartition', () => {
     expect(integerPartition(2)).toBe(2);
     expect(integerPartition(3)).toBe(3);
     expect(integerPartition(4)).toBe(5);
+    expect(integerPartition(5)).toBe(7);
+    expect(integerPartition(6)).toBe(11);
+    expect(integerPartition(7)).toBe(15);
     expect(integerPartition(8)).toBe(22);
   });
 });
diff --git a/src/algorithms/math/integer-partition/integerPartition.js b/src/algorithms/math/integer-partition/integerPartition.js
index b77cc0a3..283d9a40 100644
--- a/src/algorithms/math/integer-partition/integerPartition.js
+++ b/src/algorithms/math/integer-partition/integerPartition.js
@@ -1,5 +1,6 @@
 /**
- * @param {Number} number
+ * @param {number} number
+ * @return {number}
  */
 export default function integerPartition(number) {
   // Create partition matrix for solving this task using Dynamic Programming.