fixed class not found exception in tests, sorry