Add 'before' clause to test