Corrected bug in correcting the bug :)