Fixing a silly bug :P