re-added test class