adding directive back in