Adding a test for the API