Fix unit test

Remove reference to non-existent #define