Fixing bug from testing.