Fixed a small issue with define