fix bug with falling shapes