Små fixes
Små fixes