small fixes with animation