Bz/greg/segfault fixes