Update test mode feature