Fixes Tests

kpmartin