correcting some stuf