updated remidation code fix. tested on local setup.