Fix prior commit code issue