Fix the dimensions a little