Fixing some old tests
Getting rid of flexbox changes