Fixing a minor issue..