Fixed directive, now works!