Vital bug fixes