Fix bug from testing.