Fixed copy-paste code