Code error fix