Code error fixed