Fixing a bug/error