Bug/type fix