diff --git a/tests/site_scons/site_tools/nanopb.py b/tests/site_scons/site_tools/nanopb.py
index 4abf02a..6b82ce4 100644
--- a/tests/site_scons/site_tools/nanopb.py
+++ b/tests/site_scons/site_tools/nanopb.py
@@ -144,15 +144,12 @@
         include_dirs += ' -I' + esc(d)
 
     # when generating .pb.cpp sources, instead of pb.h generate .pb.hpp headers
-    source_extension = os.path.splitext(str(target[0]))[1]
-    header_extension = '.h' + source_extension[2:]
     nanopb_flags = env['NANOPBFLAGS']
-    if nanopb_flags:
-      nanopb_flags = '--source-extension=%s,--header-extension=%s,%s:.' % (source_extension, header_extension, nanopb_flags)
-    else:
-      nanopb_flags = '--source-extension=%s,--header-extension=%s:.' % (source_extension, header_extension)
+    source_extension = os.path.splitext(str(target[0]))[1]
+    if source_extension == '.cpp':
+        nanopb_flags += ',--source-extension=.cpp,--header-extension=.hpp'
 
-    return SCons.Action.CommandAction('$PROTOC $PROTOCFLAGS %s --nanopb_out=%s %s' % (include_dirs, nanopb_flags, srcfile),
+    return SCons.Action.CommandAction('$PROTOC $PROTOCFLAGS %s --nanopb_out=. "--nanopb_opt=%s" %s' % (include_dirs, nanopb_flags, srcfile),
                                       chdir = prefix)
 
 def _nanopb_proto_emitter(target, source, env):