add dependency
fix class reference