Fixed error in code