Minor fix, preparing to do some testing.