fixing output when testing