Fixed minor copy+paste dodgy code.