Fix entity classname