Fixed the one problem found by mutant