Update to fix scaling problem