Fixing a simple issue