Fix bugs

.