Fixing bug. Work in progress.