Fix small things

Nice