Fixed test case
Fixed crash when class doesn't exist