oops… add tests