Fixing unit test after adding licence headers