Little fix to Storage Unit