Updated code to reference correct namespace