Adding initial support for @at-root