fix simulation to use a_s as a vector