Fixed code bug.