Update database_test.go