Fix define bug