Minor modification to allow testing with geenrated Brian code