Fix directive bug