Little fix

Not complete