Fixed animation

Left out a few lines