Fix animations a bit