getClass() fix