diff --git a/db/db_impl.h b/db/db_impl.h
index 4db0cd069..a06169996 100644
--- a/db/db_impl.h
+++ b/db/db_impl.h
@@ -254,7 +254,7 @@ class DBImpl : public DB {
   Iterator* NewInternalIterator(Arena* arena,
                                 ColumnFamilyHandle* column_family = nullptr);
 
-#ifndef ROCKSDB_LITE
+#ifndef NDEBUG
   // Extra methods (for testing) that are not in the public DB interface
   // Implemented in db_impl_debug.cc
 
@@ -305,7 +305,7 @@ class DBImpl : public DB {
 
   uint64_t TEST_LogfileNumber();
 
-#endif  // ROCKSDB_LITE
+#endif  // NDEBUG
 
   // Returns the list of live files in 'live' and the list
   // of all files in the filesystem in 'candidate_files'.
diff --git a/db/db_impl_debug.cc b/db/db_impl_debug.cc
index ee437ea2b..73dca560d 100644
--- a/db/db_impl_debug.cc
+++ b/db/db_impl_debug.cc
@@ -7,7 +7,7 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file. See the AUTHORS file for names of contributors.
 
-#ifndef ROCKSDB_LITE
+#ifndef NDEBUG
 
 #include "db/db_impl.h"
 #include "util/thread_status_updater.h"
@@ -136,4 +136,4 @@ uint64_t DBImpl::TEST_LogfileNumber() {
 }
 
 }  // namespace rocksdb
-#endif  // ROCKSDB_LITE
+#endif  // NDEBUG