Fixed unknown class error