Allows margin to be passed variable integers as values, adds comments

1 files changed, 3 insertions, 3 deletions

diff --git a/packages/website/ts/@next/components/layout.tsx b/packages/website/ts/@next/components/layout.tsx
index 47d9173c3..4621d1e8e 100644
--- a/packages/website/ts/@next/components/layout.tsx
+++ b/packages/website/ts/@next/components/layout.tsx
@@ -27,7 +27,7 @@ interface SectionProps {
 interface WrapProps {
     width?: 'default' | 'full' | 'medium' | 'narrow';
     bgColor?: string;
-    margin?: number | (string | number)[];
+    margin?: number | ('large' | 'default' | number)[];
 }
 
 interface ColumnProps {
@@ -50,10 +50,10 @@ const _getColumnWidth = (args: GetColWidthArgs) => {
 };
 
 const _getPadding = (value: number | (string | number)[]) => {
-    let padding = '';
+    let padding: string = '';
 
     if (Array.isArray(value)) {
-        padding = value.map(val => PADDING_SIZES[val] || '0px').join(' ');
+        padding = value.map(val => PADDING_SIZES[val] || `${val}px`).join(' ');
     } else {
         padding = `${value}px`;
     }