Small fix update
CLEANUP