Fixes issue with directive
Fixes #8