Bug/fixes