Add tests for "extension" argument