Fix bug with size of div expansion