Minor update to scope of directive