Update entity code, now it should show