Fixes slight bug.