bug fixes

.