I will add a tests for internal classes in a bit