Fixing a silly bug