Square dragon egg

When a dragon egg teleports and switches to entity form, its model becomes a plain square with the dragon egg texture, but resumes its old shape when it converts into a block again.