fixed error from cleaning code