Small bug fix in auto_test