Fix couple of warnings