Test mode now working