Fixing issues...

