Resolved possible error in code.