diff --git a/uui-components/src/inputs/Checkbox.tsx b/uui-components/src/inputs/Checkbox.tsx
index 3498dd947b..560e524935 100644
--- a/uui-components/src/inputs/Checkbox.tsx
+++ b/uui-components/src/inputs/Checkbox.tsx
@@ -1,10 +1,8 @@
 import * as React from 'react';
-import { cx, IHasTabIndex, uuiMarkers } from '@epam/uui-core';
+import { cx, IHasTabIndex, useUuiContext, uuiMarkers } from '@epam/uui-core';
+import { Icon, uuiMod, uuiElement, isEventTargetInsideClickable, CheckboxCoreProps } from '@epam/uui-core';
+import { IconContainer } from '../layout';
 import css from './Checkbox.module.scss';
-import {
-    Icon, uuiMod, uuiElement, isEventTargetInsideClickable, CheckboxCoreProps, UuiContexts, UuiContext,
-} from '@epam/uui-core';
-import { IconContainer } from '../layout/IconContainer';
 
 export interface CheckboxProps extends CheckboxCoreProps, IHasTabIndex {
     /** Render callback for checkbox label.
@@ -26,20 +24,19 @@ export interface CheckboxProps extends CheckboxCoreProps, IHasTabIndex {
     indeterminateIcon?: Icon;
 }
 
-export class Checkbox extends React.Component<CheckboxProps> {
-    static contextType = UuiContext;
-    context: UuiContexts;
+export const Checkbox = React.forwardRef<HTMLLabelElement, CheckboxProps>((props, ref) => {
+    const context = useUuiContext();
 
-    handleChange = (e: React.SyntheticEvent<HTMLInputElement>) => {
-        !isEventTargetInsideClickable(e) && this.props.onValueChange(!this.props.value);
+    const handleChange = (e: React.SyntheticEvent<HTMLInputElement>) => {
+        !isEventTargetInsideClickable(e) && props.onValueChange(!props.value);
 
-        if (this.props.getValueChangeAnalyticsEvent) {
-            const event = this.props.getValueChangeAnalyticsEvent(!this.props.value, this.props.value);
-            this.context.uuiAnalytics.sendEvent(event);
+        if (props.getValueChangeAnalyticsEvent) {
+            const event = props.getValueChangeAnalyticsEvent(!props.value, props.value);
+            context.uuiAnalytics.sendEvent(event);
         }
     };
 
-    handleAriaCheckedValue = (indeterminate: boolean, value: boolean): boolean | 'mixed' => {
+    const handleAriaCheckedValue = (indeterminate: boolean, value: boolean): boolean | 'mixed' => {
         if (indeterminate) {
             return 'mixed';
         }
@@ -47,51 +44,46 @@ export class Checkbox extends React.Component<CheckboxProps> {
         return value == null ? false : value;
     };
 
-    render() {
-        let label = this.props.label;
-        if (this.props.renderLabel) {
-            label = this.props.renderLabel();
-        }
-        const ariaCheckedValue = this.handleAriaCheckedValue(this.props.indeterminate, this.props.value);
+    const label = props.renderLabel ? props.renderLabel() : props.label;
+    const ariaCheckedValue = handleAriaCheckedValue(props.indeterminate, props.value);
 
-        return (
-            <label
-                className={ cx(
-                    css.container,
-                    uuiElement.checkboxContainer,
-                    this.props.cx,
-                    this.props.isDisabled && uuiMod.disabled,
-                    this.props.isReadonly && uuiMod.readonly,
-                    this.props.isInvalid && uuiMod.invalid,
-                    !this.props.isReadonly && !this.props.isDisabled && uuiMarkers.clickable,
-                ) }
-                ref={ this.props.forwardedRef }
-                { ...this.props.rawProps }
+    return (
+        <label
+            className={ cx(
+                css.container,
+                uuiElement.checkboxContainer,
+                props.cx,
+                props.isDisabled && uuiMod.disabled,
+                props.isReadonly && uuiMod.readonly,
+                props.isInvalid && uuiMod.invalid,
+                !props.isReadonly && !props.isDisabled && uuiMarkers.clickable,
+            ) }
+            ref={ ref }
+            { ...props.rawProps }
+        >
+            <div
+                className={ cx(uuiElement.checkbox, (props.value || props.indeterminate) && uuiMod.checked) }
+                onFocus={ props.onFocus }
+                onBlur={ props.onBlur }
             >
-                <div
-                    className={ cx(uuiElement.checkbox, (this.props.value || this.props.indeterminate) && uuiMod.checked) }
-                    onFocus={ this.props.onFocus }
-                    onBlur={ this.props.onBlur }
-                >
-                    <input
-                        type="checkbox"
-                        onChange={ !this.props.isReadonly ? this.handleChange : undefined }
-                        disabled={ this.props.isDisabled }
-                        aria-disabled={ this.props.isDisabled || undefined }
-                        readOnly={ this.props.isReadonly }
-                        aria-readonly={ this.props.isReadonly || undefined }
-                        checked={ this.props.value || false }
-                        aria-checked={ ariaCheckedValue }
-                        required={ this.props.isRequired }
-                        aria-required={ this.props.isRequired || undefined }
-                        tabIndex={ this.props.tabIndex || this.props.isReadonly || this.props.isDisabled ? -1 : 0 }
-                        id={ this.props.id }
-                    />
-                    { this.props.value && !this.props.indeterminate && <IconContainer icon={ this.props.icon } /> }
-                    { this.props.indeterminate && <IconContainer icon={ this.props.indeterminateIcon } /> }
-                </div>
-                { label && <div className={ uuiElement.inputLabel }>{ label }</div> }
-            </label>
-        );
-    }
-}
+                <input
+                    type="checkbox"
+                    onChange={ !props.isReadonly ? handleChange : undefined }
+                    disabled={ props.isDisabled }
+                    aria-disabled={ props.isDisabled || undefined }
+                    readOnly={ props.isReadonly }
+                    aria-readonly={ props.isReadonly || undefined }
+                    checked={ props.value || false }
+                    aria-checked={ ariaCheckedValue }
+                    required={ props.isRequired }
+                    aria-required={ props.isRequired || undefined }
+                    tabIndex={ props.tabIndex || props.isReadonly || props.isDisabled ? -1 : 0 }
+                    id={ props.id }
+                />
+                { props.value && !props.indeterminate && <IconContainer icon={ props.icon } /> }
+                { props.indeterminate && <IconContainer icon={ props.indeterminateIcon } /> }
+            </div>
+            { label && <div className={ uuiElement.inputLabel }>{ label }</div> }
+        </label>
+    );
+});
diff --git a/uui-components/src/inputs/__tests__/Checkbox.test.tsx b/uui-components/src/inputs/__tests__/Checkbox.test.tsx
index f45995b7da..54e210b629 100644
--- a/uui-components/src/inputs/__tests__/Checkbox.test.tsx
+++ b/uui-components/src/inputs/__tests__/Checkbox.test.tsx
@@ -1,6 +1,6 @@
 import React from 'react';
 import { Checkbox, CheckboxProps } from '../Checkbox';
-import { render, screen, fireEvent, setupComponentForTest, userEvent } from '@epam/uui-test-utils';
+import { screen, fireEvent, setupComponentForTest, userEvent } from '@epam/uui-test-utils';
 
 async function setupCheckbox(params: Partial<CheckboxProps>) {
     const { mocks, setProps } = await setupComponentForTest<CheckboxProps>(
@@ -71,18 +71,29 @@ describe('Checkbox', () => {
         expect(getValueChangeAnalyticsEvent).toHaveBeenCalled();
     });
 
-    it('should not handle change event when readonly', () => {
+    it('should not handle change event when readonly', async () => {
         const onValueChange = jest.fn();
-        render(<Checkbox value={ false } onValueChange={ onValueChange } isReadonly />);
+        await setupCheckbox({
+            value: false,
+            onValueChange,
+            isReadonly: true,
+            label: 'Label',
+        });
         const input = screen.getByRole('checkbox');
 
         fireEvent.click(input);
         expect(onValueChange).not.toHaveBeenCalled();
     });
 
-    it('should handle focus event', () => {
+    it('should handle focus event', async () => {
         const onFocus = jest.fn();
-        render(<Checkbox value={ false } onValueChange={ jest.fn } onFocus={ onFocus } />);
+        const onValueChange = jest.fn();
+        await setupCheckbox({
+            value: false,
+            onValueChange,
+            label: 'Label',
+            onFocus,
+        });
         const input = screen.getByRole('checkbox');
 
         input.focus();
@@ -91,9 +102,15 @@ describe('Checkbox', () => {
         expect(input).toHaveFocus();
     });
 
-    it('should handle blur event', () => {
+    it('should handle blur event', async () => {
         const onBlur = jest.fn();
-        render(<Checkbox value={ false } onValueChange={ jest.fn } onBlur={ onBlur } />);
+        const onValueChange = jest.fn();
+        await setupCheckbox({
+            value: false,
+            onValueChange,
+            label: 'Label',
+            onBlur,
+        });
         const input = screen.getByRole('checkbox');
 
         input.focus();
diff --git a/uui/components/inputs/__tests__/Checkbox.test.tsx b/uui/components/inputs/__tests__/Checkbox.test.tsx
index 10a0483bc4..d9cfe60687 100644
--- a/uui/components/inputs/__tests__/Checkbox.test.tsx
+++ b/uui/components/inputs/__tests__/Checkbox.test.tsx
@@ -1,15 +1,30 @@
 import React from 'react';
-import { Checkbox } from '../Checkbox';
-import { renderer } from '@epam/uui-test-utils';
+import { renderSnapshotWithContextAsync } from '@epam/uui-test-utils';
+import { Checkbox } from '@epam/uui';
+import { ReactComponent as ActionAccountFillIcon } from '@epam/assets/icons/action-account-fill.svg';
 
-describe('Checkbox', () => {
-    it('should be rendered correctly', () => {
-        const tree = renderer.create(<Checkbox value={ null } onValueChange={ jest.fn } />).toJSON();
+describe('TestComponent', () => {
+    it('should render with minimum props', async () => {
+        const tree = await renderSnapshotWithContextAsync(<Checkbox value={ true } onValueChange={ jest.fn } />);
         expect(tree).toMatchSnapshot();
     });
 
-    it('should be rendered correctly', () => {
-        const tree = renderer.create(<Checkbox value={ null } onValueChange={ jest.fn } size="18" mode="cell" />).toJSON();
+    it('should be rendered with maximum props', async () => {
+        const tree = await renderSnapshotWithContextAsync(
+            <Checkbox
+                value={ true }
+                onValueChange={ jest.fn }
+                size="12"
+                mode="cell"
+                isDisabled={ true }
+                isInvalid={ true }
+                isReadonly={ true }
+                isRequired={ true }
+                icon={ ActionAccountFillIcon }
+                indeterminate={ true }
+                label="Test label"
+            />,
+        );
         expect(tree).toMatchSnapshot();
     });
 });
diff --git a/uui/components/inputs/__tests__/__snapshots__/Checkbox.test.tsx.snap b/uui/components/inputs/__tests__/__snapshots__/Checkbox.test.tsx.snap
index 77a3dcd8ab..c1df800019 100644
--- a/uui/components/inputs/__tests__/__snapshots__/Checkbox.test.tsx.snap
+++ b/uui/components/inputs/__tests__/__snapshots__/Checkbox.test.tsx.snap
@@ -1,37 +1,63 @@
 // Jest Snapshot v1, https://goo.gl/fbAQLP
 
-exports[`Checkbox should be rendered correctly 1`] = `
+exports[`TestComponent should be rendered with maximum props 1`] = `
 <label
-  className="container uui-checkbox-container root size-18 mode-form uui-color-primary -clickable"
+  className="container uui-checkbox-container root size-12 mode-cell uui-color-primary uui-disabled uui-readonly uui-invalid"
 >
   <div
-    className="uui-checkbox"
+    className="uui-checkbox uui-checked"
   >
     <input
-      aria-checked={false}
-      checked={false}
-      onChange={[Function]}
-      tabIndex={0}
+      aria-checked="mixed"
+      aria-disabled={true}
+      aria-readonly={true}
+      aria-required={true}
+      checked={true}
+      disabled={true}
+      readOnly={true}
+      required={true}
+      tabIndex={-1}
       type="checkbox"
     />
+    <div
+      className="container uui-icon uui-enabled"
+      style={Object {}}
+    >
+      <svg
+        className=""
+      />
+    </div>
+  </div>
+  <div
+    className="uui-input-label"
+  >
+    Test label
   </div>
 </label>
 `;
 
-exports[`Checkbox should be rendered correctly 2`] = `
+exports[`TestComponent should render with minimum props 1`] = `
 <label
-  className="container uui-checkbox-container root size-18 mode-cell uui-color-primary -clickable"
+  className="container uui-checkbox-container root size-18 mode-form uui-color-primary -clickable"
 >
   <div
-    className="uui-checkbox"
+    className="uui-checkbox uui-checked"
   >
     <input
-      aria-checked={false}
-      checked={false}
+      aria-checked={true}
+      checked={true}
       onChange={[Function]}
       tabIndex={0}
       type="checkbox"
     />
+    <div
+      className="container uui-icon uui-enabled"
+      style={Object {}}
+    >
+      <svg
+        className=""
+      />
+    </div>
   </div>
 </label>
 `;
diff --git a/uui/components/layout/__tests__/CheckboxGroup.test.tsx b/uui/components/layout/__tests__/CheckboxGroup.test.tsx
index c05a15f977..ec5c76d607 100644
--- a/uui/components/layout/__tests__/CheckboxGroup.test.tsx
+++ b/uui/components/layout/__tests__/CheckboxGroup.test.tsx
@@ -1,33 +1,30 @@
 import React from 'react';
 import { CheckboxGroup } from '../CheckboxGroup';
-import { renderer } from '@epam/uui-test-utils';
+import { renderSnapshotWithContextAsync } from '@epam/uui-test-utils';
 
 describe('CheckboxGroup', () => {
-    it('should be rendered correctly', () => {
-        const tree = renderer
-            .create(
-                <CheckboxGroup
-                    value={ null }
-                    onValueChange={ () => {} }
-                    items={ [{ id: 1, name: 'Test1' }, { id: 2, name: 'Test2' }] }
-                />,
-            )
-            .toJSON();
+    it('should be rendered correctly', async () => {
+        const tree = await renderSnapshotWithContextAsync(
+            <CheckboxGroup
+                value={ null }
+                onValueChange={ () => {} }
+                items={ [{ id: 1, name: 'Test1' }, { id: 2, name: 'Test2' }] }
+            />,
+        );
         expect(tree).toMatchSnapshot();
     });
 
-    it('should be rendered correctly with props', () => {
-        const tree = renderer
-            .create(
-                <CheckboxGroup
-                    value={ null }
-                    onValueChange={ () => {} }
-                    items={ [{ id: 1, name: 'Test1' }, { id: 2, name: 'Test2' }] }
-                    direction="horizontal"
-                    isDisabled
-                />,
-            )
-            .toJSON();
+    it('should be rendered correctly with props', async () => {
+        const tree = await renderSnapshotWithContextAsync(
+            <CheckboxGroup
+                value={ null }
+                onValueChange={ () => {} }
+                items={ [{ id: 1, name: 'Test1' }, { id: 2, name: 'Test2' }] }
+                direction="horizontal"
+                isDisabled
+            />,
+        );
+            
         expect(tree).toMatchSnapshot();
     });
 });