[Minor] fixed compile error due to wrong import to test classes