diff --git a/Makefile b/Makefile
index ee8b84a..47b347b 100644
--- a/Makefile
+++ b/Makefile
@@ -10,10 +10,10 @@ GLIB_CFLAGS := $(shell pkg-config glib-2.0 --cflags)
 
 # mariadb/mysql support
 ifeq ($(shell pkg-config --exists libmariadb && echo yes),yes)
-MYSQL_CFLAGS := $(shell pkg-config --cflags libmariadb)
+MYSQL_CFLAGS := $(shell pkg-config --cflags libmariadb) -DMARIADB=1
 MYSQL_LDFLAGS := $(shell pkg-config --libs libmariadb)
 else ifneq ($(shell which mariadb_config),)
-MYSQL_CFLAGS := $(shell mariadb_config --cflags)
+MYSQL_CFLAGS := $(shell mariadb_config --cflags) -DMARIADB=1
 MYSQL_LDFLAGS := $(shell mariadb_config --libs)
 else ifneq ($(shell which mysql_config),)
 MYSQL_CFLAGS := $(shell mysql_config --cflags)
diff --git a/medmysql.c b/medmysql.c
index 386e166..d74cf64 100644
--- a/medmysql.c
+++ b/medmysql.c
@@ -1,7 +1,12 @@
 #include <stdbool.h>
 #include <mysql.h>
+#ifdef MARIADB
+#include <mariadb/errmsg.h>
+#include <mariadb/mysqld_error.h>
+#else
 #include <mysql/errmsg.h>
 #include <mysql/mysqld_error.h>
+#endif
 #include <assert.h>
 
 #include "medmysql.h"