diff --git a/src/sled/nonstd/fsm.h b/src/sled/nonstd/fsm.h
index 20c5544..832a929 100644
--- a/src/sled/nonstd/fsm.h
+++ b/src/sled/nonstd/fsm.h
@@ -1,3 +1,7 @@
+#ifndef SLED_SLED_NONSTD_FSM_H
+#define SLED_SLED_NONSTD_FSM_H
+
+#pragma once
 /*
  * Copyright (c) 2015-2021 Thomas Kemmer
  *
@@ -22,9 +26,6 @@
  * SOFTWARE.
  */
 
-#ifndef FSMLITE_FSM_H
-#define FSMLITE_FSM_H
-
 #include <cstddef>
 #include <type_traits>
 
@@ -32,9 +33,7 @@
 #include <stdexcept>
 #endif
 
-namespace fsmlite {
-template<typename T, typename State>
-class Fsm;
+namespace sled {
 
 namespace detail {
 
@@ -452,13 +451,7 @@ private:
     };
 #endif
 };
-}// namespace fsmlite
 
-namespace sled {
-template<typename T, typename State = int>
-class Fsm : public fsmlite::Fsm<T, State> {
-public:
-    Fsm(State init_state = State()) : fsmlite::Fsm<T, State>(init_state) {}
-};
 }// namespace sled
-#endif
+
+#endif// SLED_SLED_NONSTD_FSM_H
diff --git a/src/sled/nonstd/fsm_test.cc b/src/sled/nonstd/fsm_test.cc
index f5e66f8..6b763f7 100644
--- a/src/sled/nonstd/fsm_test.cc
+++ b/src/sled/nonstd/fsm_test.cc
@@ -39,6 +39,7 @@ private:
     void OnReset(const ResetEvent &) {}
 
 public:
+    friend class sled::Fsm<FSMTest, State>;
     using TransitionTable
         = table<row<State::kIdle, StartEvent, State::kStarted, &FSMTest::OnStart, &FSMTest::StartCheck>,
                 row<State::kStarted, StopEvent, State::kEnd, &FSMTest::OnStop>,