Small code cleanup

Fixed, thanks.