Bug Fixes - Test Mode