Remove unused module from directive