Maybe we should move this to the `code_test.exs` file?