Removed SqlCodeDialog from generator class, fixed unit tests