Correct bug with .width() and .height()