Fixed error caused by copy/paste code