Fix tests
Signed-off-by:Joshua Dickerson <joshua.a.dickerson@gmail.com>