small fixesssss