fix bug by adding #define