Hotfix: Remove leftover code from testing