Added missing uses for one class