I just added a test class