Fixed leftover code error