Fixing previous bug.